/**
 * Copyright (C) 2010-2016 eBusiness Information, Excilys Group
 * Copyright (C) 2016-2020 the AndroidAnnotations project
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * Unless required by applicable law or agreed To in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package org.ohosannotations.internal.core.handler;

import com.helger.jcodemodel.AbstractJClass;
import com.helger.jcodemodel.IJAssignmentTarget;
import com.helger.jcodemodel.IJStatement;
import com.helger.jcodemodel.JBlock;
import com.helger.jcodemodel.JConditional;
import com.helger.jcodemodel.JInvocation;

import org.ohosannotations.ElementValidation;
import org.ohosannotations.OhosAnnotationsEnvironment;
import org.ohosannotations.annotations.Bean;
import org.ohosannotations.annotations.EBean;
import org.ohosannotations.annotations.NonConfigurationInstance;
import org.ohosannotations.handler.BaseAnnotationHandler;
import org.ohosannotations.handler.MethodInjectionHandler;
import org.ohosannotations.helper.InjectHelper;
import org.ohosannotations.holder.EBeanHolder;
import org.ohosannotations.holder.EComponentHolder;

import javax.lang.model.element.Element;
import javax.lang.model.element.ElementKind;
import javax.lang.model.element.TypeElement;
import javax.lang.model.type.TypeMirror;

import static com.helger.jcodemodel.JExpr._null;

/**
 * bean处理程序
 *
 * @author dev
 * @since 2021-07-22
 */
public class BeanHandler extends BaseAnnotationHandler<EComponentHolder>
    implements MethodInjectionHandler<EComponentHolder> {
    private final InjectHelper<EComponentHolder> injectHelper;

    /**
     * bean处理程序
     *
     * @param environment 环境
     */
    public BeanHandler(OhosAnnotationsEnvironment environment) {
        super(Bean.class, environment);
        injectHelper = new InjectHelper<>(validatorHelper, this);
    }

    @Override
    public void validate(Element element, ElementValidation validation) {
        injectHelper.validate(Bean.class, element, validation);
        if (!validation.isValid()) {
            return;
        }

        validatorHelper.typeOrTargetValueHasAnnotation(EBean.class, element, validation);

        validatorHelper.isNotPrivate(element, validation);
    }


    @Override
    public void process(Element element, EComponentHolder holder) {
        injectHelper.process(element, holder);
    }

    @Override
    public JBlock getInvocationBlock(EComponentHolder holder) {
        return holder.getInitBodyInjectionBlock();
    }

    @Override
    public void assignValue(JBlock targetBlock, IJAssignmentTarget fieldRef,
                            EComponentHolder holder, Element element, Element param) {
        TypeMirror typeMirror = annotationHelper.extractAnnotationClassParameter(element);
        if (typeMirror == null) {
            typeMirror = param.asType();
            typeMirror = getProcessingEnvironment().getTypeUtils().erasure(typeMirror);
        }
        String typeQualifiedName = typeMirror.toString();
        AbstractJClass injectedClass =
            getJClass(annotationHelper.generatedClassQualifiedNameFromQualifiedName(typeQualifiedName));
        JInvocation beanInstance = injectedClass
            .staticInvoke(EBeanHolder.GET_INSTANCE_METHOD_NAME)
            .arg(holder.getContextRef());

        TypeElement declaredEBean = getProcessingEnvironment().getElementUtils().getTypeElement(typeQualifiedName);
        if (declaredEBean != null) {
            EBean annotation = declaredEBean.getAnnotation(EBean.class);
            if (annotation.scope() != EBean.Scope.Singleton && annotation.scope() != EBean.Scope.Ability) {
                beanInstance.arg(holder.getRootFragmentRef());
            }
        }

        IJStatement assignment = fieldRef.assign(beanInstance);
        if (param.getKind() == ElementKind.FIELD) {
            boolean hasNonConfigurationInstanceAnnotation =
                element.getAnnotation(NonConfigurationInstance.class) != null;
            if (hasNonConfigurationInstanceAnnotation) {
                JConditional conditional = targetBlock._if(fieldRef.eq(_null()));
                conditional._then().add(assignment);
                assignment = conditional;
            }
        }

        targetBlock.add(assignment);
    }

    @Override
    public void validateEnclosingElement(Element element, ElementValidation valid) {
        validatorHelper.enclosingElementHasEnhancedComponentAnnotation(element, valid);
    }
}
